package com.onlygx.dva.common.utils;
import cn.hutool.core.io.FileUtil;

import java.io.*;
import java.text.DecimalFormat;

/**
 * 瞳见科技
 * 文件信息获取或生成
 * @author : GaoXiang
 * 时间 : 2021/12/23
 * 介绍 :
 */
public class FileInfoUtils {

    //private static final String TAG = FileSizeUtils.class.getSimpleName();

    public static final int TYPE_B = 1;//获取文件大小单位为B的double值
    public static final int TYPE_KB = 2;//获取文件大小单位为KB的double值
    public static final int TYPE_MB = 3;//获取文件大小单位为MB的double值
    public static final int TYPE_GB = 4;//获取文件大小单位为GB的double值


    /**
     * 获取文件后缀名
     * @param fileName 文件名
     * @return 后缀名 不带 “.”
     */
    public static String getSuffix(String fileName){
        return fileName.substring(fileName.lastIndexOf(".")+1);
    }


    public static void createDir(String path){
        File file = new File(path);
        if  (!file .exists()  && !file .isDirectory())
        {
            boolean success = file .mkdir();
            System.out.println("文件夹创建状态 " + success + " : " + path);
        }
    }

    /**
     * 获取网络文件名
     * 仅适用于具有绝对地址的文件
     */
    public static String getWebFileName(String url){
        return url.substring(url.lastIndexOf("/")+1);
    }

    /**
     * 获取文件指定文件的指定单位的大小
     *
     * @param filePath 文件路径
     * @param sizeType 获取大小的类型1为B、2为KB、3为MB、4为GB
     * @return double值的大小
     */
    public static double getFileOrFilesSize(String filePath, int sizeType) {
        File file = new File(filePath);
        long blockSize = 0;
        try {
            if (file.isDirectory()) {
                blockSize = getFileSizes(file);
            } else {
                blockSize = getFileSize(file);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return FormatFileSize(blockSize, sizeType);
    }

    /**
     * 调用此方法自动计算指定文件或指定文件夹的大小
     *
     * @param filePath 文件路径
     * @return 计算好的带B、KB、MB、GB的字符串
     */
    public static String getAutoFileOrFilesSize(String filePath) {
        File file = new File(filePath);
        long blockSize = 0;
        try {
            if (file.isDirectory()) {
                blockSize = getFileSizes(file);
            } else {
                blockSize = getFileSize(file);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return formatFileSize(blockSize);
    }

    /**
     * 获取指定文件大小
     */
    private static long getFileSize(File file) throws Exception {
        long size = 0;
        if (file.exists()) {
            FileInputStream fis;
            fis = new FileInputStream(file);
            size = fis.available();
        } else {
            //boolean success = file.createNewFile();
            return size;
        }
        return size;
    }

    /**
     * 获取指定文件夹
     */
    private static long getFileSizes(File f) throws Exception {
        long size = 0;
        File[] fileList = f.listFiles();
        assert fileList != null;
        for (File file : fileList) {
            if (file.isDirectory()) {
                size = size + getFileSizes(file);
            } else {
                size = size + getFileSize(file);
            }
        }
        return size;
    }

    /**
     * 转换文件大小
     */
    public static String formatFileSize(long fileS) {
        DecimalFormat df = new DecimalFormat("#.00");
        String fileSizeString;
        String wrongSize = "0B";
        if (fileS == 0) {
            return wrongSize;
        }
        if (fileS < 1024) {
            fileSizeString = df.format((double) fileS) + "B";
        } else if (fileS < 1048576) {
            fileSizeString = df.format((double) fileS / 1024) + "KB";
        } else if (fileS < 1073741824) {
            fileSizeString = df.format((double) fileS / 1048576) + "MB";
        } else {
            fileSizeString = df.format((double) fileS / 1073741824) + "GB";
        }
        return fileSizeString;
    }

    /**
     * 转换文件大小,指定转换的类型
     */
    private static double FormatFileSize(long fileS, int sizeType) {
        DecimalFormat df = new DecimalFormat("#.00");
        double fileSizeLong = 0;
        switch (sizeType) {
            case TYPE_B:
                fileSizeLong = Double.parseDouble(df.format((double) fileS));
                break;
            case TYPE_KB:
                fileSizeLong = Double.parseDouble(df.format((double) fileS / 1024));
                break;
            case TYPE_MB:
                fileSizeLong = Double.parseDouble(df.format((double) fileS / 1048576));
                break;
            case TYPE_GB:
                fileSizeLong = Double.parseDouble(df.format((double) fileS / 1073741824));
                break;
            default:
                break;
        }
        return fileSizeLong;
    }

    public static void main(String args[]) throws Exception{
        String rootPath = FileUtil.file(".", "").getCanonicalPath();
        FileInfoUtils.createDir(rootPath);
        String fullPath = rootPath + "download\\" + "aaa" + ".json";
        File file = new File(fullPath);
        if (!file.getParentFile().exists()) { // 如果⽗⽬录不存在，创建⽗⽬录
            file.getParentFile().mkdirs();
        }
        if (file.exists()) { // 如果已存在,删除旧⽂件
            file.delete();
        }
        file.createNewFile();
    }

    /**
     * ⽣成.json格式⽂件
     */
    public static String createJsonFile(String jsonString, String fileName) {
        // 标记⽂件⽣成是否成功
        boolean flag = true;
        String fullPath = "";
        // ⽣成json格式⽂件
        try {
            String rootPath = FileUtil.file(".", "/").getCanonicalPath();
            FileInfoUtils.createDir(rootPath);
            // 拼接⽂件完整路径
            fullPath = rootPath + "/download/" + fileName + ".json";
            // 保证创建⼀个新⽂件
            File file = new File(fullPath);
            if (!file.getParentFile().exists()) { // 如果⽗⽬录不存在，创建⽗⽬录
                file.getParentFile().mkdirs();
            }
            if (file.exists()) { // 如果已存在,删除旧⽂件
                file.delete();
            }
            file.createNewFile();
            // 将格式化后的字符串写⼊⽂件
            Writer write = new OutputStreamWriter(new FileOutputStream(file), "UTF-8");
            write.write(jsonString);
            write.flush();
            write.close();
        } catch (Exception e) {
            flag = false;
            e.printStackTrace();
        }
        // 返回是否成功的标记
        return fullPath;
    }
}
